package com.lu.springbootinit.mq;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.DeliverCallback;

import java.io.IOException;

public class MultiConsumer {
    // 声明队列名称为"multi_queue"
    private static final String TASK_QUEUE_NAME = "multi_queue";

    public static void main(String[] argv) throws Exception {
        // 创建一个新的连接工厂
        ConnectionFactory factory = new ConnectionFactory();
        // 设置连接工厂的主机地址
        factory.setHost("localhost");
        factory.setUsername("lu");
        factory.setPassword("Lqb326706");
        factory.setVirtualHost("/lel");
        // 从工厂获取一个新的连接
        final Connection connection = factory.newConnection();

        for (int i = 0; i < 2; i++) {
            final Channel channel = connection.createChannel();

            channel.queueDeclare(TASK_QUEUE_NAME, true, false, false, null);
            System.out.println(" [*] Waiting for messages. To exit press CTRL+C");
            // 设置预取计数为 1，这样RabbitMQ就会在给消费者新消息之前等待先前的消息被确认
            channel.basicQos(1);

            int finalI = i;
            DeliverCallback deliverCallback = (consumerTag, delivery) -> {
                String message = new String(delivery.getBody(), "UTF-8");

                try {
                    System.out.println(" [x] Received '" + "编号:"+ finalI + ":" + message + "'");
                    Thread.sleep(20000);
                    channel.basicAck(delivery.getEnvelope().getDeliveryTag(),false);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } finally {
                    System.out.println(" [x] Done");
                    channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
                }
            };
            channel.basicConsume(TASK_QUEUE_NAME, false, deliverCallback, consumerTag -> { });
        }



    }

    //初始单独连接的算法
    private static void WayFirst(Connection connection) throws IOException {
        // 从连接获取一个新的通道
        final Channel channel = connection.createChannel();
        // 声明一个队列,并设置属性:队列名称,持久化,非排他,非自动删除,其他参数;如果队列不存在,则创建它
        channel.queueDeclare(TASK_QUEUE_NAME, true, false, false, null);
        // 在控制台打印等待消息的提示信息
        System.out.println(" [*] Waiting for messages. To exit press CTRL+C");
        // (这个先注释)设置预取计数为1，这样RabbitMQ就会在给消费者新消息之前等待先前的消息被确认
        channel.basicQos(1);

        // 创建消息接收回调函数,以便接收消息
        DeliverCallback deliverCallback = (consumerTag, delivery) -> {
            // 将接收到的消息转为字符串
            String message = new String(delivery.getBody(), "UTF-8");

            try {
                // (放到try里)打印出接收到的消息
                System.out.println(" [x] Received '" + message + "'");
                // 处理工作,模拟处理消息所花费的时间,机器处理能力有限(接收一条消息,20秒后再接收下一条消息)
                Thread.sleep(20000);
                // (不用doWork)模拟消息处理工作
                // doWork(message);
            } catch (InterruptedException e) {
                // 模拟处理消息所花费的时间
                e.printStackTrace();
            } finally {
                // 打印出完成消息处理的提示
                System.out.println(" [x] Done");
                // 手动发送应答,告诉RabbitMQ消息已经被处理
                channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
            }
        };
        // 开始消费消息,传入队列名称,是否自动确认,投递回调和消费者取消回调
        channel.basicConsume(TASK_QUEUE_NAME, false, deliverCallback, consumerTag -> { });
    }
    // (不用doWork)用于模拟消息处理的函数，消息中的每一个'.'字符都会让线程暂停一秒钟
     private static void doWork(String task) {
       for (char ch : task.toCharArray()) {
           if (ch == '.') {
               try {
                   // 暂停一秒钟
                   Thread.sleep(1000);
               } catch (InterruptedException _ignored) {
                   // 在被中断的情况下重新设置线程的中断状态
                   Thread.currentThread().interrupt();
               }
           }
       }
     }
}



